package com.qixiucoding.util;

import org.apache.commons.lang3.ObjectUtils;

import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 * @Author longhj
 * @Description TODO
 * @Date 2021/08/28 星期六 16:00
 */
public class CollectionUtil {
    /**
     * 获取输入集合的特定字段信息
     *
     * @param coll
     * @param args
     * @return
     */
    public static Map<String, List<String>> getAttrVal(List<Map<String, String>> coll, String... args) {
        Map<String, List<String>> resultMap = new HashMap() {{
            for (String arg : args) {
                put(arg, new LinkedList<String>());
            }
        }};
        if (args.length < 1) {
            return resultMap;
        }
        for (String arg : args) {
            coll.stream().forEach(map -> {
                if (ObjectUtils.isNotEmpty(map.get(arg))) {
                    resultMap.get(arg).add(map.get(arg));
                }
            });
        }
        return resultMap;
    }
}
